import tensorflow as tf

tensor = tf.constant([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

# 索引获取单个元素
print(tensor[0, 1]) # 2，和列表一样，寻找最外层第1个元素内的第2个元素

# 切片操作
row = tensor[1] # 取出第2行
col = tensor[:, 1] # 取出第2列
sub = tensor[0:2,1:] # 取出第0行到第1行，第2列到最后一列
print(row)
print(col)
print(sub)